#pragma once

#include <iostream>
#include <fstream>
#include <string>
#include <atomic>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <unistd.h>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    class TimeUtil
    {
    public:
        static std::string GetTimeStamp()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec);
        }
        // 获取毫秒时间戳
        static std::string GetTimeMs()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 10000);
        }
    };

    static std::string temp_path = "./temp/";

    class PathUtil
    {
    private:
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }

    public:
        // 编译时，需要有的临时文件
        // 构建源文件路径 + 后缀的完整文件名
        // 123 -> ./temp/123.cpp
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }
        // 构建可执行程序的完整路径 + 后缀的完整文件名
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }
        static std::string CompilerError(const std::string &file_name)
        {
            return AddSuffix(file_name, "compile_error");
        }

        // 运行时, 需要有的临时文件
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };

    class FileUtil
    {
    public:
        // 检查文件是否存在
        static bool IsFileExists(const std::string &path_name)
        {
            struct stat statbuf;
            // 系统调用
            if (stat(path_name.c_str(), &statbuf /*无用*/) == 0)
            {
                // 获取文件属性成功，文件已经存在
                return true;
            }
            return false;
        }
        static std::string UniqFileName()
        {
            static std::atomic_uint id(0); // 全局独一份，不要每次进来重定义
            // 毫秒级时间戳 + 原子性递增唯一值(锁 / atomic): 来保证唯一性
            std::string ms = TimeUtil::GetTimeMs();
            std::string uniq_id = std::to_string(id);
            id++; // 使用完更新 id

            return ms + "_" + uniq_id;
        }
        // 向文件中写
        static bool WriteFile(const std::string &file_name, const std::string &content)
        {
            std::ofstream outfile(file_name); // 向谁中写
            if (!outfile.is_open())
            {
                return false;
            }
            outfile.write(content.c_str(), content.size());
            outfile.close();
            return true;
        }
        // 从文件中读
        static bool ReadFile(const std::string &file_name, std::string *content, bool keep = false)
        {
            (*content).clear();
            std::ifstream in(file_name);
            if (!in.is_open())
            {
                return false;
            }

            std::string line;
            // getline: 不保存行分隔符，有些时候需要保存 \n
            // getline 内部重载了强制类型转化
            while (std::getline(in, line))
            {
                (*content) += line;
                (*content) += (keep ? "\n" : ""); // keep 意味是否需要换行
            }
            in.close();
            return true;
        }
    };

    class StringUtil
    {
    public:
        /************************************
         * str: 输入型，目标要切分的字符串
         * target: 输出型，保存切分完毕的结果
         * sep: 指定的分割符
         * *********************************/
        static void SpiltString(const std::string &str, std::vector<std::string> *target,
                                const std::string &sep)
        {
            boost::split((*target), str, boost::is_any_of(sep),
                         boost::algorithm::token_compress_on/*压缩重复分隔符为一个, eg: "   "->" "*/);
        }
    };
}